from selenium.webdriver.remote.webdriver import WebDriver


class AddImgs:

    def add_img(self, driver,imgs):
        self.imgs.append(driver.get_screenshot_as_base64())
        return True, self
    def __init__(self,driver):
        # self.driver = driver
        self.imgs = []
        self.add_img(driver,  self.imgs)
